package com.xzc.lease.common.utils;

import com.xzc.lease.common.exception.LeaseException;
import com.xzc.lease.common.result.ResultCodeEnum;
import io.jsonwebtoken.*;
import io.jsonwebtoken.security.Keys;

import javax.crypto.SecretKey;
import java.util.Date;

/**
 * @author 徐生
 * 2024/10/12
 * description：创建JWT令牌的工具类
 */
public class JwtUtil {

    private static final SecretKey secretKey = Keys.hmacShaKeyFor("M0PKKI6pYGVWWfDZw90a0lTpGYX1d4AQ".getBytes());

    //创建Jwt令牌
    public static String createToken(Long userId, String username) {
        return Jwts.builder()
                //设置 JWT Payload部分
                //设置过期时间   System.currentTimeMillis()单位为毫秒，即设置1小时过期时间 (方便测试时间调长，范围超过Integer在最后加个L)
                .setExpiration(new Date(System.currentTimeMillis() + 3600000 * 24 * 30L))
                //设置主题
                .setSubject("LOGIN_USER")
                //自定义字段
                .claim("userId", userId)
                .claim("username", username)
                //设置JWT signature部分
                //设置签名  secretKey为密钥， 采用HS256加密算法
                .signWith(secretKey, SignatureAlgorithm.HS256)
                .compact();
    }

    //校验Jwt令牌
    public static Claims parseToken(String token) {
        if (token == null) {
            throw new LeaseException(ResultCodeEnum.ADMIN_LOGIN_AUTH);
        }

        try {
            JwtParser jwtParser = Jwts.parserBuilder().setSigningKey(secretKey).build();
            //解析token
            Jws<Claims> claimsJws = jwtParser.parseClaimsJws(token);
            //得到 jwt的payload部分，用于获取自定义属性
            return claimsJws.getBody();
        } catch (ExpiredJwtException e) { //token过期异常
            throw new LeaseException(ResultCodeEnum.TOKEN_EXPIRED);
        } catch (JwtException e) {
            throw new LeaseException(ResultCodeEnum.TOKEN_INVALID);
        }
    }

    //测试jwt生成
    public static void main(String[] args) {
        //生成的jwt，需要在每个 swagger 分组中配置请求头
        //参数名称：access-token     参数值：运行main()生成     参数类型：header
        System.out.println(createToken(2L, "user"));
        //获取app端 token
        System.out.println(createToken(8L, "17607033291"));
    }
}
